import numpy as np

def convert_ort_type_2_np(ort_data_type):
    #logger.info("convert_ort_type_2_np")
    
    types = {
        1 : np.float32,
        2 : np.uint8,
        3 : np.int8,
        4 : np.uint16,
        5 : np.int16,
        6 : np.int32,
        7 : np.int64,
        8 : "",  #string
        9 : np.bool_,
        10 : np.float16,
        11 : np.float64,
        12 : np.uint32,
        13 : np.uint64,
        14 : np.complex64,
        15 : np.complex_,
        16 : ""
    }

    return types.get(ort_data_type, None)

def get_data_list(dtype, init):
    data_list = []

    if dtype == 1: #float
        data_list = init.float_data

    if dtype == 2: #uint8
        data_list = init.int32_data

    if dtype == 3: #int8
        data_list = init.int32_data    

    if dtype == 4: #uint16
        data_list = init.int32_data

    if dtype == 5: #int16
        data_list = init.int32_data

    if dtype == 6: #int32
        data_list = init.int32_data

    if dtype == 7: #int64
        data_list = init.int64_data

    if dtype == 8: #string
        data_list = init.string_data       

    if dtype == 9: #bool
        data_list = init.int32_data    

    if dtype == 10: #float16
        data_list = init.int32_data
         
    if dtype == 11: #double
        data_list = init.double_data      

    if dtype == 12: #uint32
        data_list = init.uint64_data  

    if dtype == 13: #uint64
        data_list = init.uint64_data

    return data_list   